from torchvision import models
from torchvision import transforms
from PIL import Image
import torch


def main():

    a = torch.tensor([[1,2],[2,3],[3,4]])

    print("a.shape=", a.shape)
    print("a=", a)

    b = a.unsqueeze(0)
    print("b.shape=", b.shape)
    print("b=", b)





if __name__ == '__main__':
    main()
